import PyInstaller.__main__
import os
import shutil

def build_exe():
    # 清理之前的构建文件
    if os.path.exists('build'):
        shutil.rmtree('build')
    if os.path.exists('dist'):
        shutil.rmtree('dist')
    
    # PyInstaller参数
    args = [
        'main.py',  # 主脚本
        '--onefile',  # 打包成单个exe文件
        '--noconsole',  # 不显示控制台窗口
        '--name=client-app',  # 输出文件名
        '--clean',  # 清理临时文件
        # 添加图标（如果有）
        # '--icon=icon.ico',
    ]
    
    # 运行PyInstaller
    PyInstaller.__main__.run(args)
    
    print("构建完成！exe文件位于 dist/client-app.exe")
